#include "OLED_Font.h"

#define OLED_ADDR 0x3C

/**
 * @brief  I2C 写数据
 * @param  addr 地址
 * @param  data 数据
 * @return 0 | 1
 */
uint8_t I2C_WriteByte(uint8_t addr, uint8_t data) {

  uint8_t buff[2] = {0};

  buff[0] = addr;

  buff[1] = data;

  /* Wait for I2C to be Idle */

  while (!(

      DL_I2C_getControllerStatus(I2C_0_INST) & DL_I2C_CONTROLLER_STATUS_IDLE))
    ;

  /* Send the packet to the controller.

   * This function will send Start + Stop automatically.

   */

  DL_I2C_startControllerTransfer(I2C_0_INST, OLED_ADDR,

                                 DL_I2C_CONTROLLER_DIRECTION_TX, 2);

  DL_I2C_fillControllerTXFIFO(I2C_0_INST, &buff[0], 2);

  /* Poll until the Controller writes all bytes */

  while (DL_I2C_getControllerStatus(I2C_0_INST) &

         DL_I2C_CONTROLLER_STATUS_BUSY_BUS);

  /* Trap if there was an error */

//   if (DL_I2C_getControllerStatus(I2C_0_INST) &

//       DL_I2C_CONTROLLER_STATUS_ERROR) {

    /* LED will remain high if there is an error */

    // __BKPT(0);
//   }

  return 0;
}

/**
 * @brief  OLED写命令
 * @param  Command 要写入的命令
 * @return 无
 */
void OLED_WriteCommand(uint8_t Command) {
  I2C_WriteByte(0x00, Command);
}

/**
 * @brief  OLED写数据
 * @param  Data 要写入的数据
 * @return 无
 */
void OLED_WriteData(uint8_t Data) {
  I2C_WriteByte(0x40, Data);
}

/**
 * @brief  OLED设置光标位置
 * @param  Y 以左上角为原点，向下方向的坐标，范围：0~7
 * @param  X 以左上角为原点，向右方向的坐标，范围：0~127
 * @return 无
 */
void OLED_SetCursor(uint8_t Y, uint8_t X) {
  OLED_WriteCommand(0xB0 | Y);                 // 设置Y位置
  OLED_WriteCommand(0x10 | ((X & 0xF0) >> 4)); // 设置X位置高4位
  OLED_WriteCommand(0x00 | (X & 0x0F));        // 设置X位置低4位
}

/**
 * @brief  OLED清屏
 * @param  无
 * @return 无
 */
void OLED_Clear(void) {
  uint8_t i, j;
  for (j = 0; j < 8; j++) {
    OLED_SetCursor(j, 0);
    for (i = 0; i < 128; i++) {
      OLED_WriteData(0x00);
    }
  }
}

/**
 * @brief  OLED显示一个字符
 * @param  Line 行位置，范围：1~4
 * @param  Column 列位置，范围：1~16
 * @param  Char 要显示的一个字符，范围：ASCII可见字符
 * @return 无
 */
void OLED_ShowChar(uint8_t Line, uint8_t Column, char Char) {
  uint8_t i;
  OLED_SetCursor((Line - 1) * 2, (Column - 1) * 8); // 设置光标位置在上半部分
  for (i = 0; i < 8; i++) {
    OLED_WriteData(OLED_F8x16[Char - ' '][i]); // 显示上半部分内容
  }
  OLED_SetCursor((Line - 1) * 2 + 1,
                 (Column - 1) * 8); // 设置光标位置在下半部分
  for (i = 0; i < 8; i++) {
    OLED_WriteData(OLED_F8x16[Char - ' '][i + 8]); // 显示下半部分内容
  }
}

/**
 * @brief  OLED显示字符串
 * @param  Line 起始行位置，范围：1~4
 * @param  Column 起始列位置，范围：1~16
 * @param  String 要显示的字符串，范围：ASCII可见字符
 * @return 无
 */
void OLED_ShowString(uint8_t Line, uint8_t Column, char *String) {
  uint8_t i;
  for (i = 0; String[i] != '\0'; i++) {
    OLED_ShowChar(Line, Column + i, String[i]);
  }
}

/**
 * @brief  OLED次方函数
 * @return 返回值等于X的Y次方
 */
uint32_t OLED_Pow(uint32_t X, uint32_t Y) {
  uint32_t Result = 1;
  while (Y--) {
    Result *= X;
  }
  return Result;
}

/**
 * @brief  OLED显示数字（十进制，正数）
 * @param  Line 起始行位置，范围：1~4
 * @param  Column 起始列位置，范围：1~16
 * @param  Number 要显示的数字，范围：0~4294967295
 * @param  Length 要显示数字的长度，范围：1~10
 * @return 无
 */
void OLED_ShowNum(uint8_t Line, uint8_t Column, uint32_t Number,
                  uint8_t Length) {
  uint8_t i;
  for (i = 0; i < Length; i++) {
    OLED_ShowChar(Line, Column + i,
                  Number / OLED_Pow(10, Length - i - 1) % 10 + '0');
  }
}

/**
 * @brief  OLED显示数字（十进制，带符号数）
 * @param  Line 起始行位置，范围：1~4
 * @param  Column 起始列位置，范围：1~16
 * @param  Number 要显示的数字，范围：-2147483648~2147483647
 * @param  Length 要显示数字的长度，范围：1~10
 * @return 无
 */
void OLED_ShowSignedNum(uint8_t Line, uint8_t Column, int32_t Number,
                        uint8_t Length) {
  uint8_t i;
  uint32_t Number1;
  if (Number >= 0) {
    OLED_ShowChar(Line, Column, '+');
    Number1 = Number;
  } else {
    OLED_ShowChar(Line, Column, '-');
    Number1 = -Number;
  }
  for (i = 0; i < Length; i++) {
    OLED_ShowChar(Line, Column + i + 1,
                  Number1 / OLED_Pow(10, Length - i - 1) % 10 + '0');
  }
}

/**
 * @brief  OLED显示数字（十六进制，正数）
 * @param  Line 起始行位置，范围：1~4
 * @param  Column 起始列位置，范围：1~16
 * @param  Number 要显示的数字，范围：0~0xFFFFFFFF
 * @param  Length 要显示数字的长度，范围：1~8
 * @return 无
 */
void OLED_ShowHexNum(uint8_t Line, uint8_t Column, uint32_t Number,
                     uint8_t Length) {
  uint8_t i, SingleNumber;
  for (i = 0; i < Length; i++) {
    SingleNumber = Number / OLED_Pow(16, Length - i - 1) % 16;
    if (SingleNumber < 10) {
      OLED_ShowChar(Line, Column + i, SingleNumber + '0');
    } else {
      OLED_ShowChar(Line, Column + i, SingleNumber - 10 + 'A');
    }
  }
}

/**
 * @brief  OLED显示数字（二进制，正数）
 * @param  Line 起始行位置，范围：1~4
 * @param  Column 起始列位置，范围：1~16
 * @param  Number 要显示的数字，范围：0~1111 1111 1111 1111
 * @param  Length 要显示数字的长度，范围：1~16
 * @return 无
 */
void OLED_ShowBinNum(uint8_t Line, uint8_t Column, uint32_t Number,
                     uint8_t Length) {
  uint8_t i;
  for (i = 0; i < Length; i++) {
    OLED_ShowChar(Line, Column + i,
                  Number / OLED_Pow(2, Length - i - 1) % 2 + '0');
  }
}

/**
 * @brief  OLED初始化
 * @param  无
 * @return 无
 */
void OLED_Init(void) {
  uint32_t i, j;

  for (i = 0; i < 1000; i++) // 上电延时
  {
    for (j = 0; j < 1000; j++)
      ;
  }

  // OLED_I2C_Init(); // 端口初始化

  OLED_WriteCommand(0xAE); // 关闭显示

  OLED_WriteCommand(0xD5); // 设置显示时钟分频比/振荡器频率
  OLED_WriteCommand(0x80);

  OLED_WriteCommand(0xA8); // 设置多路复用率
  OLED_WriteCommand(0x3F);

  OLED_WriteCommand(0xD3); // 设置显示偏移
  OLED_WriteCommand(0x00);

  OLED_WriteCommand(0x40); // 设置显示开始行

  OLED_WriteCommand(0xA1); // 设置左右方向，0xA1正常 0xA0左右反置

  OLED_WriteCommand(0xC8); // 设置上下方向，0xC8正常 0xC0上下反置

  OLED_WriteCommand(0xDA); // 设置COM引脚硬件配置
  OLED_WriteCommand(0x12);

  OLED_WriteCommand(0x81); // 设置对比度控制
  OLED_WriteCommand(0xCF);

  OLED_WriteCommand(0xD9); // 设置预充电周期
  OLED_WriteCommand(0xF1);

  OLED_WriteCommand(0xDB); // 设置VCOMH取消选择级别
  OLED_WriteCommand(0x30);

  OLED_WriteCommand(0xA4); // 设置整个显示打开/关闭

  OLED_WriteCommand(0xA6); // 设置正常/倒转显示

  OLED_WriteCommand(0x8D); // 设置充电泵
  OLED_WriteCommand(0x14);

  OLED_WriteCommand(0xAF); // 开启显示

  OLED_Clear(); // OLED清屏
}
